package com.lb.pdm;

import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import com.baomidou.mybatisplus.generator.fill.Column;

import java.util.Collections;
import java.util.Scanner;

/**
 * MybatisPlus代码生成器配置类
 * 支持交互式配置和批量生成
 * 
 * @author pdm
 * @since 2024-01-15
 */
public class CodeGeneratorConfig {

    /**
     * 数据库连接配置
     */
    private static final String URL = "jdbc:mysql://8.140.225.7:4406/pdm?useUnicode=true&characterEncoding=UTF-8&autoReconnect=true&useSSL=false&allowPublicKeyRetrieval=true&serverTimezone=UTC";
    private static final String USERNAME = "pdm-user";
    private static final String PASSWORD = "123456";

    /**
     * 项目路径配置
     */
    private static final String PROJECT_PATH = System.getProperty("user.dir");
    private static final String JAVA_PATH = PROJECT_PATH + "/src/main/java";
    private static final String MAPPER_XML_PATH = PROJECT_PATH + "/src/main/resources/mapper";

    /**
     * 包名配置
     */
    private static final String PARENT_PACKAGE = "com.lb.pdm";

    /**
     * 作者信息
     */
    private static final String AUTHOR = "pdm";

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        
        System.out.println("=== MybatisPlus代码生成器 ===");
        System.out.println("请选择生成模式：");
        System.out.println("1. 生成所有表");
        System.out.println("2. 生成指定表");
        System.out.println("3. 生成单个表");
        System.out.print("请输入选择 (1-3): ");
        
        int choice = scanner.nextInt();
        scanner.nextLine(); // 消费换行符
        
        switch (choice) {
            case 1:
                generateAllTables();
                break;
            case 2:
                generateSpecificTables(scanner);
                break;
            case 3:
                generateSingleTable(scanner);
                break;
            default:
                System.out.println("无效选择！");
                break;
        }
        
        scanner.close();
    }

    /**
     * 生成所有预定义的表
     */
    private static void generateAllTables() {
        String[] tableNames = {
            "sys_user",
            "person_info", 
            "household_info",
            "household_member",
            "birth_registration",
            "household_transfer",
            "family_reunion",
            "floating_population",
            "key_personnel",
            "certificate_info",
            "certificate_application",
            "sys_dict",
            "operation_log"
        };
        
        System.out.println("开始生成所有表的代码...");
        executeGeneration(tableNames);
    }

    /**
     * 生成指定的多个表
     */
    private static void generateSpecificTables(Scanner scanner) {
        System.out.println("可用的表名：");
        System.out.println("sys_user, person_info, household_info, household_member,");
        System.out.println("birth_registration, household_transfer, family_reunion,");
        System.out.println("floating_population, key_personnel, certificate_info,");
        System.out.println("certificate_application, sys_dict, operation_log");
        System.out.print("请输入要生成的表名（用逗号分隔）: ");
        
        String input = scanner.nextLine();
        String[] tableNames = input.split(",");
        
        // 清理表名（去除空格）
        for (int i = 0; i < tableNames.length; i++) {
            tableNames[i] = tableNames[i].trim();
        }
        
        System.out.println("开始生成指定表的代码...");
        executeGeneration(tableNames);
    }

    /**
     * 生成单个表
     */
    private static void generateSingleTable(Scanner scanner) {
        System.out.print("请输入要生成的表名: ");
        String tableName = scanner.nextLine().trim();
        
        System.out.println("开始生成表 " + tableName + " 的代码...");
        executeGeneration(new String[]{tableName});
    }

    /**
     * 执行代码生成
     * 
     * @param tableNames 表名数组
     */
    private static void executeGeneration(String[] tableNames) {
        FastAutoGenerator.create(URL, USERNAME, PASSWORD)
                // 全局配置
                .globalConfig(builder -> {
                    builder.author(AUTHOR) // 设置作者
                            .enableSwagger() // 开启 swagger 模式
                            .outputDir(JAVA_PATH) // 指定输出目录
                            .outputDir(JAVA_PATH) // 指定输出目录
                            .dateType(DateType.ONLY_DATE) // 时间类型对应策略
                            .commentDate("yyyy-MM-dd") // 注释日期格式
                            .disableOpenDir(); // 禁止打开输出目录
                })
                // 包配置
                .packageConfig(builder -> {
                    builder.parent(PARENT_PACKAGE) // 设置父包名
                            .entity("entity") // 设置实体类包名
                            .mapper("mapper") // 设置 Mapper 接口包名
                            .service("service") // 设置 Service 接口包名
                            .serviceImpl("service.impl") // 设置 Service 实现类包名
                            .controller("controller") // 设置 Controller 包名
                            .pathInfo(Collections.singletonMap(OutputFile.xml, MAPPER_XML_PATH)); // 设置mapperXml生成路径
                })
                // 策略配置
                .strategyConfig(builder -> {
                    builder.addInclude(tableNames) // 设置需要生成的表名
                            .addTablePrefix("t_", "c_") // 设置过滤表前缀
                            
                            // 实体类策略配置
                            .entityBuilder()
                            .enableLombok() // 开启 lombok 模型
                            .enableTableFieldAnnotation() // 开启生成实体时生成字段注解
                            .enableChainModel() // 开启链式模型
                            .naming(NamingStrategy.underline_to_camel) // 数据库表映射到实体的命名策略
                            .columnNaming(NamingStrategy.underline_to_camel) // 数据库表字段映射到实体的命名策略
                            .idType(IdType.AUTO) // 主键类型
                            .formatFileName("%s") // 实体类文件名格式
                            .addTableFills(new Column("create_time", FieldFill.INSERT)) // 自动填充配置
                            .addTableFills(new Column("update_time", FieldFill.INSERT_UPDATE))
                            .addTableFills(new Column("created_at", FieldFill.INSERT))
                            .addTableFills(new Column("updated_at", FieldFill.INSERT_UPDATE))
                            
                            // Mapper策略配置
                            .mapperBuilder()
                            .enableMapperAnnotation() // 开启 @Mapper 注解
                            .enableBaseResultMap() // 启用 BaseResultMap 生成
                            .enableBaseColumnList() // 启用 BaseColumnList
                            .formatMapperFileName("%sMapper") // Mapper文件名格式
                            .formatXmlFileName("%sMapper") // XML文件名格式
                            
                            // Service策略配置
                            .serviceBuilder()
                            .formatServiceFileName("%sService") // 格式化 service 接口文件名称
                            .formatServiceImplFileName("%sServiceImpl") // 格式化 service 实现类文件名称
                            
                            // Controller策略配置
                            .controllerBuilder()
                            .enableRestStyle() // 开启生成@RestController 控制器
                            .enableHyphenStyle() // 开启驼峰转连字符
                            .formatFileName("%sController"); // Controller文件名格式
                })
                // 注入配置
                .injectionConfig(consumer -> {
                    // 可以在这里添加自定义配置
                })
                // 模板引擎配置，使用 Freemarker 引擎模板
                .templateEngine(new FreemarkerTemplateEngine())
                .execute();

        System.out.println("=== 代码生成完成！===");
        System.out.println("生成路径：" + JAVA_PATH);
        System.out.println("Mapper XML路径：" + MAPPER_XML_PATH);
        System.out.println("生成的表：" + String.join(", ", tableNames));
        
        System.out.println("\n生成的文件结构：");
        for (String tableName : tableNames) {
            String className = toCamelCase(tableName);
            System.out.println("├── " + className + ".java (实体类)");
            System.out.println("├── " + className + "Mapper.java (Mapper接口)");
            System.out.println("├── " + className + "Service.java (Service接口)");
            System.out.println("├── " + className + "ServiceImpl.java (Service实现)");
            System.out.println("├── " + className + "Controller.java (Controller)");
            System.out.println("└── " + className + "Mapper.xml (Mapper XML)");
            System.out.println();
        }
    }

    /**
     * 将下划线命名转换为驼峰命名
     */
    private static String toCamelCase(String str) {
        StringBuilder result = new StringBuilder();
        String[] parts = str.split("_");
        
        for (String part : parts) {
            if (part.length() > 0) {
                result.append(Character.toUpperCase(part.charAt(0)))
                      .append(part.substring(1).toLowerCase());
            }
        }
        
        return result.toString();
    }
}